Score Centering Stabilizes Off-policy Reinforcement Learning
分数中心化稳定离线策略强化学习
🧭 LLM 强化学习训练稳定性校正
Together AI
提出加性 score centering 校正项,抵消训练与推理引擎间的漂移,稳定 LLM 的 RL 训练
📌 工作参考价值(4 点)
- 在 RL 微调推荐/对话策略时,若部署推理引擎与训练引擎存在量化、KV cache 压缩或异步参数不一致,可在计算优势或回报时减去漂移项(如训练与推理引擎在同样本上的平均输出差异),低成本稳定训练,避免策略崩溃。
- Score centering 是加性修正,可与 importance sampling 等乘法修正叠加;在离线策略数据来自旧策略或不同推理引擎时,组合使用可能比单用 IS 更稳健,尤其 mismatch 严重时。
- 对于生成式推荐(用 LLM 直接生成 item ID/文案),RL 训练中 reward 常由规则或模型给出,容易受采样分布偏移影响;考虑在 reward 或 advantage 上做 centering 来抵消环境漂移,可提升训练稳定性。
- 工程上:漂移估计可通过在固定验证集上比较训练和推理引擎的 logprob/奖励差异得到,开销很小;可作为 RL 训练 pipeline 的常规监控与校正组件。